package com.stardata.liangshisq.tcslmon.util;

import org.apache.shiro.authz.AuthorizationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

/**
 * @author Samson Shu
 * @email shush@stardata.top
 * @date 2020/6/5 17:00
 */
@RestControllerAdvice(value = {"com.stardata.liangshisq.tcslmon.controller"})
public class RRExceptionHandler {
  private Logger logger = LoggerFactory.getLogger(getClass());

  /**
   * 自定义异常
   */
  @ExceptionHandler(RRException.class)
  public R handleRRException(RRException e) {
    R r = R.error(e.getMsg());
    r.put("code", e.getCode());
    r.put("msg", e.getMessage());

    return r;
  }

  @ExceptionHandler(DuplicateKeyException.class)
  public R handleDuplicateKeyException(DuplicateKeyException e) {
    logger.error(e.getMessage(), e);
    return R.error("数据库中已存在该记录");
  }

  @ExceptionHandler(AuthorizationException.class)
  public R handleAuthorizationException(AuthorizationException e) {
    logger.error(e.getMessage(), e);
    return R.error("没有权限，请联系管理员授权");
  }

  @ExceptionHandler(Exception.class)
  public R handleException(Exception e) {
    logger.error(e.getMessage(), e);
    return R.error();
  }


  @ExceptionHandler(ApiRRException.class)
  public Object handleApiRRException(ApiRRException e) {
    int code = e.getStatus().value();
    HttpStatus status = HttpStatus.valueOf(code);
    R result = R.custom(status, e.getErrmsg());
    result.put("errcode",code);
    result.put("errmsg", e.getErrmsg());
    return result;
  }
}
